# 写一个装饰器，为存钱和取钱添加安全保护

# 新人小钱，为取钱函数添加 输入密码功能，密码争取才能取钱
def password_check(fn):
    def wrap_func(name, x):  # 包裹的函数
        # 验证密码，如果密码不正确，则不调用原来的旧函数
        password = input('请输入密码：')
        if password == '123456':
            fn(name, x)
        else:
            print('密码验证失败！')
    return wrap_func
# 小张写的两个在银行存钱和取钱的两个函数
@password_check
def save_money(name, x):
    print(name, '存钱', x, '元')

@password_check
def withdraw(name, x):
    print(name, '取钱', x, '元')

# 小李调用小张写的程序，为前台提供服务
save_money('小王', 200)
save_money('小赵', 500)
withdraw('小吴', 800)
withdraw('小林', 200)
